package com.lcm.algorithm.datatest.dp;

import java.util.*;

/**
 * @description: 139. 单词拆分
 *
 * 给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。
 *
 * 说明：
 *     拆分时可以重复使用字典中的单词。
 *     你可以假设字典中没有重复的单词。
 *     不需要包含字典中所有的单词。
 *
 * 示例 1：
 *
 * 输入: s = "leetcode", wordDict = ["leet", "code"]
 * 输出: true
 * 解释: 返回 true 因为 "leetcode" 可以被拆分成 "leet code"。
 *
 * 示例 2：
 *
 * 输入: s = "applepenapple", wordDict = ["apple", "pen"]
 * 输出: true
 * 解释: 返回 true 因为 "applepenapple" 可以被拆分成 "apple pen apple"。
 *      注意你可以重复使用字典中的单词。
 *
 * 示例 3：
 *
 * 输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
 * 输出: false
 * @author: lcm
 * @create: 2020-04-28 21:22
 **/

public class WordBreak {

    public static void main(String[] args){
        WordBreak wordBreak=new WordBreak();
        System.out.println(wordBreak.wordBreak("leetcode", Arrays.asList(new String[]{"leet","code"})));

    }

    /**
     * 运行时间：55ms
     * 占用内存：12360k
     * 1.递归暴力。
     * 2.递归记忆化搜索。
     */
    public boolean wordBreak(String s, List<String> wordDict) {
        HashSet<Integer> set=new HashSet<>();
        return help(s,new HashSet<>(wordDict),0,set);
    }

    boolean help(String s, Set<String> set,int begin,Set<Integer> dp){
        if(begin==s.length()){
            return true;
        }
        if(dp.contains(begin)) return false;
        for(int i=begin+1;i<=s.length();i++){
            String str=s.substring(begin,i);
            if(set.contains(str)&&help(s,set,i,dp)){
                return true;
            }
        }
        dp.add(begin);
        return false;
    }

    public List<String> wordBreakTwo(String s, List<String> wordDict) {
        HashSet<Integer> set=new HashSet<>();
        List<String> list=new ArrayList<>();
        helpTwo(s,new HashSet<>(wordDict),"",set,list);
        return list;
    }

    void helpTwo(String s, Set<String> set,String word,Set<Integer> dp,List<String> list){
        int begin=word.length();
        if(begin==s.length()){
            list.add(word);
            return;
        }
        if(dp.contains(begin)) {
            return;
        }
        for(int i=begin+1;i<=s.length();i++){
            String str=s.substring(begin,i);
            if(set.contains(str)){
                helpTwo(s,set,word+" "+str,dp,list);
            }
        }
        dp.add(begin);
        return;
    }

}
